#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
创建题库Excel模板文件
"""

import pandas as pd
import os

def create_template():
    # 创建模板数据
    data = {
        '题目': ['单选题：以下哪个是茶叶的主要产地？', '多选题：以下哪些属于绿茶？', '判断题：红茶是发酵茶。'],
        '选项A': ['福建', '西湖龙井', '是'],
        '选项B': ['广东', '碧螺春', '否'],
        '选项C': ['云南', '铁观音', ''],
        '选项D': ['浙江', '毛尖', ''],
        '答案': ['A,D', 'A,B,D', '是'],
        '图片': ['', '', '']
    }
    
    # 创建DataFrame
    df = pd.DataFrame(data)
    
    # 确保目录存在
    template_dir = './images/questions'
    if not os.path.exists(template_dir):
        os.makedirs(template_dir)
    
    # 保存为Excel文件
    template_path = os.path.join(template_dir, '题库.xlsx')
    df.to_excel(template_path, index=False, engine='openpyxl')
    
    print(f"模板文件已创建: {os.path.abspath(template_path)}")

if __name__ == '__main__':
    create_template()
